package com.yugao.fintech.antelope.auth.utils;

import cn.hutool.core.convert.Convert;
import com.yugao.fintech.antelope.base.config.GlobalProperties;
import com.yugao.fintech.antelope.base.model.constants.SecurityCons;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.Map;
import java.util.Objects;

/**
 * Jwt工具类
 */
@Component
public class JwtUtils {
    @Autowired
    private GlobalProperties properties;
    public static String secret = SecurityCons.SECRET;

    @PostConstruct
    public void init() {
        secret = properties.getJwtSecret();
    }

    /**
     * 从数据声明生成令牌
     *
     * @return 令牌
     */
    public static String createToken(Map<String, Object> claims, String randomSecret) {
        return Jwts.builder().setClaims(claims).signWith(SignatureAlgorithm.HS512, secret).compact();
    }

    /**
     * 从令牌中获取数据声明
     *
     * @param token 令牌
     * @return 数据声明
     */
    public static Claims parseToken(String token) {
        try {
            return Jwts.parser().setSigningKey(secret).parseClaimsJws(token).getBody();
        } catch (Exception e) {
            return null;
        }
    }


    /**
     * 根据令牌获取用户ID
     *
     * @param token 令牌
     * @return 用户ID
     */
    public static String getUserId(String token) {
        Claims claims = parseToken(token);
        if (Objects.isNull(claims)) {
            return null;
        }
        return getValue(claims, SecurityCons.USER_ID);
    }

    /**
     * 根据身份信息获取用户ID
     *
     * @param claims 身份信息
     * @return 用户ID
     */
    public static String getUserId(Claims claims) {
        return getValue(claims, SecurityCons.USER_ID);
    }


    /**
     * 根据身份信息获取键值
     *
     * @param claims 身份信息
     * @param key    键
     * @return 值
     */
    public static String getValue(Claims claims, String key) {
        return Convert.toStr(claims.get(key), "");
    }
}
